Debug and test SRJaxAgent - #23

PySR's jax callable indexes X by fit-time column, and JAX clamps
out-of-range indices instead of raising. Running a model distilled from
a window_size=10 history under the default window_size=5 therefore read
the wrong columns, collapsed the expression to a constant, and pinned
every station to TX for the whole rollout: constant collisions, zero
throughput, flat plots.

SRJaxAgent now compares feature_names_in_ against the feature count the
simulation supplies and names the window size to use.

The sr-run target passes that window size explicitly, and defaults to 10
epochs since the all_* plots draw one point per epoch and a single-epoch
run renders as blank axes.

Also adds the unit test for history_reshape, covering the row and column
ordering the CSV columns and the agent both depend on.

Co-Authored-By: Claude Opus 5 <noreply@anthropic.com>